#pragma once
#include <iostream>
#include <string>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "../comm/util.hpp"
#include "../comm/log.hpp"

//该头文件负责进行代码的编译
namespace ns_compiler
{
    // 引入路径拼接、日志功能
    using namespace ns_util;
    using namespace ns_log;
    class Compiler
    {
    public:
        Compiler()
        {}
        ~Compiler()
        {}

        //编译函数，返回值代表编译成功与否，参数为要编译的文件名(无后缀)
        static bool Compile(const std::string& file_name)
        {
            pid_t pid = fork();
            if(pid < 0)
            {
                //std::cerr << "子进程创建失败！" << std::endl;
                LOG(ERROR) << "内部错误：创建子进程失败！" <<std::endl;
                return false;
            }
            else if(pid == 0)
            {   
                //child:调用编译器完成对代码的编译工作

                //创建编译错误文件
                umask(0);//权限掩码清零，方便接下来设置文件权限
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(),O_CREAT | O_WRONLY,0644); 
                if(_stderr < 0)
                {
                    //std::cerr << "创建错误文件失败！" << std::endl;
                    LOG(WARNING) << "未能成功形成CompilerError文件！" << std::endl;
                    exit(1);
                }
                //g++编译错误时会把错误信息打印到cerr,重定向标准错误到_stderr，以便编译错误时直接打印至文件中而非cerr
                dup2(_stderr,2);//本该打印到2号文件描述符的东西重定向至_stderr

                //程序替换不会影响进程的文件描述符表！！
                //在temp目录下构建出三个文件：.cpp文件、.exe文件、.stderr错误文件
                //注意此处需要手动定义COMPILE_ONLINE宏，因为题目的header与tail已经拼接完成，这里如果不定义会报错(文件找不到)
                //g++ -o target src -D COMPILER_ONLINE -std=c++11 
                execlp("g++","g++","-o",PathUtil::Exe(file_name).c_str(),PathUtil::Src(file_name).c_str(),"-D","COMPILER_ONLINE","-std=c++11",nullptr/*不要忘记最后以空结尾*/);
            
                //程序替换失败则退出
                LOG(ERROR) << "启动编译器g++失败，可能是参数错误！" << std::endl;
                exit(2);
            }
            else
            {
                //father

                //等待子进程，不关心其退出情况
                waitpid(pid,nullptr,0);
                //走到此处，子进程编译完成，要处理结果
                //可执行程序是否生成则代表编译是否通过
                if(FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << " 编译成功"  << std::endl;
                    return true;
                }
                LOG(ERROR) << "编译失败，未形成可执行程序！" << std::endl;
                return false;
            }
        }
    };
}